# export CUDA_VISIBLE_DEVICES=2
CUDA_LAUNCH_BLOCKING=1 python train.py
